"""
__author__ = '霍格沃兹测试开发学社'
__desc__ = '更多测试开发技术探讨，请访问：https://ceshiren.com/t/topic/15860'
"""
from dao.base_dao import BaseDao
from orm_domain.suite import Suite
from server import db_session


class SuiteDao(BaseDao):

    def save(self, suite: Suite):
        self.add(suite)


    def get_by_suite_id(self, _id) -> Suite:
        s_obj = self.query(Suite).filter_by(id=_id).first()
        return s_obj
